﻿# -*- coding: utf-8 -*-
# @Date   : 2020/01/02
# @File   : TinyCompanyLoanCnt_2D.py
# @Author : yanhaopeng

'''eof
name:小额贷款公司未结清贷款笔数
code:TinyCompanyLoanCnt_2D
tableName:
columnName:
groups:个人征信
dependencies:IC_DSJ_2D
type:常用指标
datasourceType:在线指标
description:
eof'''


import sys

reload(sys)
sys.setdefaultencoding('utf-8')

null_type_list = ['', ' ', None, 'null', 'Null', 'NULL', '/', '-', '--']

def getTable(data, tableName, occurTimes):
    '''
        获取表信息
        :param data:表上层的字典对象
        :param tableName:表的名字
        :param occurTimes:表出现的次数，如果是0代表此表可以不出现，如果是1代表此表必须出现
        :return:该表全部数据
        '''
    table = data.get(tableName)
    if occurTimes == 0:
        if table is None or table == "":
            return [{}]
        elif isinstance(table, list) is False or len(table) == 0:
            return None
        else:
            return data.get(tableName)
    if occurTimes == 1:
        if table == "":
            return [{}]
        elif table is None or isinstance(table, list) is False or len(table) == 0:
            return None
        elif len(table) == 0 or table == "":
            return [{}]
        else:
            return data.get(tableName)

def isfloat(x):
    '''
    判断x为float类型且大于等于0
    '''
    try:
        if isinstance(float(x), float) and float(x) >= 0:
            if str(x) in ['inf', 'infinity', 'INF', 'INFINITY', 'True']:
                return False
            else:
                return True
        else:
            return False
    except:
        return False



'''
1、Factor1 = 信贷交易信息明细(PD01)--基本信息段(PD01A)--账户类型(PD01AD01)为D1非循环贷，
            且最新表现信息段(PD01B).账户状态(PD01BD01)为【1-正常，2-逾期，4-呆账，6-担保物不足，7-强制平仓，8-司法追偿】
            且余额(PD01BJ01)【大于0】时；
            且信贷交易信息明细(PD01).基本信息段(PD01A).业务管理机构类型(PD01AD02)为【51-小额贷款公司】的次数；

2、Factor2 = 信贷交易信息明细(PD01)--基本信息段(PD01A).账户类型(PD01AD01)为R1循环贷，
            且最新表现信息段(PD01B).账户状态(PD01BD01)为【1-正常，2-逾期，4-呆账，5-银行止付,6-担保物不足，8-司法追偿】
            且余额(PD01BJ01)【大于0】时；
            且信贷交易信息明细(PD01).基本信息段(PD01A).业务管理机构类型(PD01AD02)为【51-小额贷款公司】的次数；

3、Factor3 = 信贷交易信息明细(PD01)--基本信息段(PD01A)--账户类型(PD01AD01)为R4循环贷分账户，
            且最新表现信息段(PD01B).账户状态(PD01BD01)为【1-正常，2-逾期，4-呆账，6-担保物不足，8-司法追偿】
            且余额(PD01BJ01)【大于0】时；
            且信贷交易信息明细(PD01).基本信息段(PD01A).业务管理机构类型(PD01AD02)为【51-小额贷款公司】的次数；

【指标结果】
Factor =  Factor1 + Factor2 + Factor3
'''

def TinyCompanyLoanCnt():
    data = IC_DSJ_2D['data']
    #借贷账户信息   [1..1]
    PDA = getTable(data, 'PDA', 1)
    if PDA is None:
        return u'缺失值'
    elif PDA == [{}]:
        return 0
    else:
        result = []
        # 借贷账户信息单元   [0..*]
        PD01 = getTable(PDA[0], 'PD01', 0)
        if PD01 is None:
            return u'缺失值'
        elif PD01 == [{}]:
            return 0
        else:
            for i in PD01:
                try:
                    # 基本信息段   [1..1]
                    PD01A = getTable(i, 'PD01A', 1)
                    # 最新表现信息段   [1..1]
                    PD01B = getTable(i, 'PD01B', 1)
                    if PD01A is None or PD01B is None:
                        result.append(None)
                    else:
                        # 账户类型   [1..1]
                        PD01AD01 = PD01A[0].get('PD01AD01')
                        # 业务机构类型   [1..1]
                        PD01AD02 = PD01A[0].get('PD01AD02')
                        #账户状态   [1..1]
                        PD01BD01 = PD01B[0].get('PD01BD01')
                        #余额   [1..1]
                        PD01BJ01 = PD01B[0].get('PD01BJ01')
                        if PD01AD01 in null_type_list \
                            or PD01AD02 in null_type_list \
                            or PD01BD01 in null_type_list \
                            or PD01BJ01 in null_type_list:
                            result.append(None)
                        else:
                            # 51-小额贷款公司
                            if PD01AD02 == '51':
                                # D1-非循环贷账户；1-正常，2-逾期，4-呆账，6-担保物不足，7-强制平仓，8-司法追偿
                                if PD01AD01 == 'D1' and PD01BD01 in ['1','2','4','6','7','8'] and isfloat(PD01BJ01) and float(PD01BJ01) > 0:
                                    result.append(1)
                                # R1-循环贷账户；1-正常，2-逾期，4-呆账，5-银行止付,6-担保物不足，8-司法追偿
                                elif PD01AD01 == 'R1' and PD01BD01 in ['1','2','4','5','6','8'] and isfloat(PD01BJ01) and float(PD01BJ01) > 0:
                                    result.append(1)
                                # R1-循环贷分账户；1-正常，2-逾期，4-呆账，6-担保物不足，8-司法追偿，8-司法追偿
                                elif PD01AD01 == 'R4' and PD01BD01 in ['1','2','4','6','8'] and isfloat(PD01BJ01) and float(PD01BJ01) > 0:
                                    result.append(1)
                                else:
                                    result.append(0)
                            else:
                                result.append(0)
                except:
                    result.append(None)
                    continue

            result = filter(lambda x: x is not None, result)
            if len(result) == 0:
                return u'缺失值'
            else:
                return sum(result)


result = TinyCompanyLoanCnt()

